package com.captjack.core.configuration;

import com.jfinal.kit.PathKit;
import com.jfinal.kit.Prop;
import com.jfinal.kit.PropKit;
import com.jfinal.plugin.activerecord.dialect.MysqlDialect;
import com.jfinal.plugin.activerecord.generator.Generator;
import com.jfinal.plugin.hikaricp.HikariCpPlugin;

import javax.sql.DataSource;

/**
 * @author Capt Jack
 * @date 2018/5/11
 */
public class GeneratorCode {

    /**
     * 配置数据源
     *
     * @return 数据源
     */
    private static DataSource getDataSource() {
        // 加载配置文件
        Prop p = PropKit.use("data-source.txt");
        // 创建c3p0连接
        HikariCpPlugin hikariCpPlugin = new HikariCpPlugin(p.get("jdbcUrl"), p.get("user"), p.get("password"));
        // 启用hikariCpPlugin插件
        hikariCpPlugin.start();
        return hikariCpPlugin.getDataSource();
    }

    public static void main(String[] args) {
        // base model 所使用的包名
        String baseModelPackageName = "com.captjack.core.model.base";
        // base model 文件保存路径
        String baseModelOutputDir = PathKit.getWebRootPath() + "/src/main/java/com/captjack/wechat/model/base";

        // model 所使用的包名 (MappingKit 默认使用的包名)
        String modelPackageName = "com.captjack.core.model";
        // model 文件保存路径 (MappingKit 与 DataDictionary 文件默认保存路径)
        String modelOutputDir = baseModelOutputDir + "/..";
        // 创建生成器
        Generator generator = new Generator(getDataSource(), baseModelPackageName, baseModelOutputDir, modelPackageName, modelOutputDir);
        // 设置数据库方言
        generator.setDialect(new MysqlDialect());
        // 添加不需要生成的表名
        generator.addExcludedTable("adv");
        // 设置是否在 Model 中生成 dao 对象
        generator.setGenerateDaoInModel(true);
        // 设置是否生成字典文件
        generator.setGenerateDataDictionary(false);
        // 设置需要被移除的表名前缀用于生成modelName。例如表名 "osc_user"，移除前缀 "osc_"后生成的model名为 "User"而非 OscUser
        generator.setRemovedTableNamePrefixes("x_");
        // 执行生成
        generator.generate();
    }

}
